# 合并排序数组
# 问题描述：合并两个升序的整数数组A和B，形成一个新的数组，新的数组也要有序。
# 问题示例：办理入A=[1]，B=[1]，输出[1,1]，返回合并后的数组。输入A=[1,2,3,4]，B=[2,4,5,6]，返回合并所有元素后的数组
class Solution:
    def mergeSortedArray(self, A, B):
        i, j = 0, 0
        c = []
        while i < len(A) and j < len(B):
            if A[i] < B[j]:
                c.append(A[i])
                i += 1
            else:
                c.append(B[j])
                j += 1
        while i < len(A):
            c.append(A[i])
            i += 1
        while j < len(B):
            c.append(B[j])
            j += 1
        return c


if __name__ == '__main__':
    A = [1, 2, 3, 4]
    B = [2, 4, 5, 6]
    solution = Solution()
    print(solution.mergeSortedArray(A, B))
